from pydantic import BaseModel
from typing import Optional


class AIModelResponse(BaseModel):
    id: str
    name: str
    description: str
    type: str
    
    class Config:
        orm_mode = True


class AIChatRequest(BaseModel):
    message: str
    modelId: Optional[str] = "qwen-turbo"
    
    class Config:
        orm_mode = True


class AIChatResponse(BaseModel):
    response: str
    modelId: str
    success: str
    
    class Config:
        orm_mode = True